import torch
import torch_npu

x=torch.randn(2,2).npu()
y=torch.randn(2,2).npu()
z=x.mm(y)
print(z)